from tool import get_completion


if __name__ == '__main__':
    customer_email = """
    嗯呐，我现在可是火冒三丈，我那个搅拌机盖子竟然飞了出去，把我厨房的墙壁都溅上了果汁！
    更糟糕的是，保修条款可不包括清理我厨房的费用。
    伙计，赶紧给我过来！
    """
    # 普通话 + 平静、尊敬的语调
    style = """正式普通话 \
    用一个平静、尊敬、有礼貌的语调
    """

    # 要求模型根据给出的语调进行转化
    prompt = f"""把由三个反引号分隔的文本\
    翻译成一种{style}风格。
    文本: ```{customer_email}```
    """
    print("提示：", prompt)

    response = get_completion(prompt)
    print(response)
